from langchain_openai import ChatOpenAI
from dotenv import load_dotenv
import os

load_dotenv()

DEEPSEEK_BASE_URL = os.getenv("DEEPSEEK_BASE_URL")
DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
DEEPSEEK_MODEL = os.getenv("DEEPSEEK_MODEL")

# Description: Configuration file
class Config:
    def __init__(self):
        """
        Initializes the configuration for the agent
        """
        self.BASE_LLM = ChatOpenAI(
            model=DEEPSEEK_MODEL,
            temperature=0.2,
            base_url=DEEPSEEK_BASE_URL,
            api_key=DEEPSEEK_API_KEY
        )
        self.FACTUAL_LLM = ChatOpenAI(
            model=DEEPSEEK_MODEL,
            temperature=0.0,
            base_url=DEEPSEEK_BASE_URL,
            api_key=DEEPSEEK_API_KEY
        )
        self.DEBUG = False